/*
 * (c) 2020-2021 nitz — chris marc dailey https://cmd.wtf
 * (c) 2015-2017 Marcos Del Sol Vives
 *
 * SPDX-License-Identifier: MIT
 */

#include "nfc3d/keygen.h"
#include "nfc3d/drbg.h"
#include <assert.h>
#include <stdio.h>
#include <string.h>

#if defined(NRF52832_XXAA)
// string.h doesn't want to give memccpy to me on this platform.
void* memccpy(void* __restrict, const void* __restrict, int, size_t);
#endif // NRF52840_XXAA

void nfc3d_keygen_prepare_seed(const nfc3d_keygen_masterkeys* baseKeys, const uint8_t* baseSeed, uint8_t* output, size_t* outputSize)
{
	assert(baseKeys != NULL);
	assert(baseSeed != NULL);
	assert(output != NULL);
	assert(outputSize != NULL);

	uint8_t* start = output;

	// 1: Copy whole type string
	output = memccpy(output, baseKeys->typeString, '\0', sizeof(baseKeys->typeString));

	// 2: Append (16 - magicBytesSize) from the input seed
	size_t leadingSeedBytes = 16 - baseKeys->magicBytesSize;
	memcpy(output, baseSeed, leadingSeedBytes);
	output += leadingSeedBytes;

	// 3: Append all bytes from magicBytes
	memcpy(output, baseKeys->magicBytes, baseKeys->magicBytesSize);
	output += baseKeys->magicBytesSize;

	// 4: Append bytes 0x10-0x1F from input seed
	memcpy(output, baseSeed + 0x10, 16);
	output += 16;

	// 5: Xor last bytes 0x20-0x3F of input seed with AES XOR pad and append them
	unsigned int i;
	for (i = 0; i < 32; i++)
	{
		output[i] = baseSeed[i + 32] ^ baseKeys->xorPad[i];
	}
	output += 32;

	*outputSize = output - start;
}

void nfc3d_keygen(const nfc3d_keygen_masterkeys* baseKeys, const uint8_t* baseSeed, nfc3d_keygen_derivedkeys* derivedKeys)
{
	uint8_t preparedSeed[NFC3D_DRBG_MAX_SEED_SIZE];
	size_t preparedSeedSize;

	nfc3d_keygen_prepare_seed(baseKeys, baseSeed, preparedSeed, &preparedSeedSize);
	nfc3d_drbg_generate_bytes(baseKeys->hmacKey, sizeof(baseKeys->hmacKey), preparedSeed, preparedSeedSize, (uint8_t*)derivedKeys, sizeof(*derivedKeys));
}
